import torch

# 随机生成数据
# 导入torch库，用于后续的张量操作
import torch

# 生成一个4x5的张量，元素为0到9之间的整数，用于后续的示例操作
data = torch.randint(0, 10, [4, 5])
# 打印生成的张量
print(data)

# 打印张量的第一行，即索引为0的行
print(data[0])
# 打印张量的第一列，即索引为0的列，这里使用了冒号表示选取所有行
print(data[:, 0])
# 打印张量中第二列（索引为2）元素大于5的所有行，这里使用了布尔索引
print(data[data[:, 2] > 5])
# 打印张量中第二行（索引为1）元素大于5的所有列，这里同样使用了布尔索引
print(data[:, data[1] > 5])
